import random
import pygame

# 定义屏幕大小的常量
SCREEN_RECT = pygame.Rect(0,0,480,700)
# 定义刷新帧率的常量
FRAME_PER_SEC = 60
# 创建敌机的定时期常量
CREATE_ENEMY_EVENT = pygame.USEREVENT
# 英雄发射子弹事件
HERO_FIRE_EVENT = pygame.USEREVENT + 1


class GameSprite(pygame.sprite.Sprite):
    """飞机大战游戏精灵"""

    def __init__(self, image_name,speed = 1,speed2 = 0):
        # 调用父类的初始化方法
        super().__init__()
        # 定义对象的属性
        self.image = pygame.image.load(image_name)
        self.rect = self.image.get_rect()
        self.speed = speed
        self.speed2 = speed2
        
    def update(self):
        # 在屏幕垂直方向上移动
        self.rect.y += self.speed


class Background(GameSprite):
    """游戏背景精灵"""

    def __init__(self,is_alt = False):
        # 调用父类方法实现精灵创建(image/rect/speed)
        super().__init__("python演练（飞机大战）/images/background.png")
        # 判断是否是交替图像，如果是，需要设置初始位置
        if is_alt:
            self.rect.y = -self.rect.height
   
    def update(self):
        # 调用父类方法实现
        super().update() 
        # 判断图像是否移出屏幕
        if self.rect.y >= SCREEN_RECT.height:
            self.rect.y = -self.rect.height


class Enemy(GameSprite):

    def __init__(self):
        # 调用父类方法，创建敌机精灵，同时指定敌机图片
        super().__init__("python演练（飞机大战）/images/enemy1.png")
        # 指定敌机的初始随机速度 1-3
        self.speed = random.randint(2,3)
        # 指定敌机的初始随机位置
        self.rect.bottom = 0
        MAX_x = SCREEN_RECT.width - self.rect.width
        self.rect.x = random.randint(0,MAX_x)

    def update(self):
        # 调用父类方法，保持垂直方向飞行
        super().update()
        # 判断是否飞出屏幕，如果是需要从精灵组删除敌机
        if self.rect.y >= SCREEN_RECT.height:
            # kill方法可以将所有精灵从精灵组移除，精灵就会被自动销毁
            self.kill()

    def __del__(self):
        pass


class Hero(GameSprite):
    """英雄精灵"""

    def __init__(self):
        super().__init__("python演练（飞机大战）/images/me1.png",0)
        # 设置英雄初始位置
        self.rect.centerx = SCREEN_RECT.centerx
        self.rect.bottom = SCREEN_RECT.bottom - 120
        # 创建子弹精灵组
        self.bullets = pygame.sprite.Group()

    def update(self):
        self.rect.x += self.speed
        self.rect.y += self.speed2
        # 控制英雄不能离开屏幕
        if self.rect.x < 0:
            self.rect.x = 0
        elif self.rect.right > SCREEN_RECT.right:
            self.rect.right =SCREEN_RECT.right

    def fire(self):
        # 连续发射子弹
        for i in (0,1,2):
            bullet = Bullet()  # 创建子弹精灵
            # 设置子弹精灵位置
            bullet.rect.bottom = self.rect.y - i * 20
            bullet.rect.centerx = self.rect.centerx
            self.bullets.add(bullet)  # 将精灵添加到精灵组


class Bullet(GameSprite):
    """子弹精灵"""

    def __init__(self):
        super().__init__("python演练（飞机大战）/images/bullet1.png",-4)

    def update(self):
        super().update()
        if self.rect.bottom < 0:
            self.kill()
    
    def __del__(self):
        pass